/*
求解一个给定的方程，将x以字符串 "x=#value" 的形式返回。该方程仅包含 '+' ， '-' 操作，变量 x 和其对应系数。

如果方程没有解，请返回 "No solution" 。如果方程有无限解，则返回 “Infinite solutions” 。

题目保证，如果方程中只有一个解，则 'x' 的值是一个整数。

 

示例 1：

输入: equation = "x+5-3+x=6+x-2"
输出: "x=2"
示例 2:

输入: equation = "x=x"
输出: "Infinite solutions"
示例 3:

输入: equation = "2x=x"
输出: "x=0"
 

提示:

3 <= equation.length <= 1000
equation 只有一个 '='.
equation 方程由整数组成，其绝对值在 [0, 100] 范围内，不含前导零和变量 'x' 。 

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/solve-the-equation
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
*/

#include "../stdc++.h"

using namespace std;

class Solution {
public:
    string solveEquation(string equation) {
        int factor = 0; // 变量的系数
        int val = 0; // 常量的值
        int index = 0;
        int n = equation.size();
        int sign1 = 1; // 等式左边默认系数为正
        while (index < n) {
            if (equation[index] == '=') {
                sign1 = -1; // 等式右边默认系数为负
                ++index;
                continue;
            }
            int sign2 = sign1;
            int number = 0;
            bool valid = false; // 记录number是否有效
            if (equation[index] == '-' || equation[index] == '+') {
                // 去掉前面的符号
                sign2 = (equation[index] == '-') ? -sign1 : sign1;
                ++index;
            }
            while (index < n && isdigit(equation[index])) {
                number = number * 10 + (equation[index] - '0');
                ++index;
                valid = true;
            }
            if (index < n && equation[index] == 'x') {
                // 变量
                factor += valid ? sign2 * number : sign2;
                ++index;
            } else {
                // 数值
                val += sign2 * number;
            }
        }
        if (factor == 0) {
            return val == 0? "Infinite solutions" : "No solution";
        }
        return string("x=") + to_string(-val / factor);
    }
};
